package org.example.threadlocaltest.demos.configure;

import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

//一个拦截器，拦截请求，把请求的时间记录下来，并将请求头中的 token 拿出来
public class MyInterceptor implements HandlerInterceptor {

    @Override//在请求处理之前进行调用（Controller方法调用之前）
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        //记录请求时间
        long startTime = System.currentTimeMillis();
        request.setAttribute("startTime", startTime);
        //将请求头中的 token 拿出来
//        String token = request.getHeader("token");
        //从 Authorization 中获取 token 并去除 Bearer 字符串
        String token = request.getHeader("Authorization");
        if (token != null && token.startsWith("Bearer ")) {
            token = token.substring(7);
        }

        if (token == null) {
            throw new Exception("无 token，请重新登录");
        }

        //将 token 放入 ThreadLocal 中
        TokenUtils.setToken(token);
        return true;
    }

    @Override//请求处理之后进行调用，但是在视图被渲染之前（Controller方法调用之后）
    public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception {
        //获取请求时间
        long startTime = (long) request.getAttribute("startTime");
        //计算请求处理时间
        long endTime = System.currentTimeMillis();
        System.out.println("本次请求处理时间为：" + (endTime - startTime) + "ms");
    }

    @Override//在整个请求结束之后被调用，也就是在 DispatcherServlet 渲染了对应的视图之后执行
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {
        //移除 ThreadLocal 中的 token
        System.out.println("清除 token" );
        TokenUtils.clearToken();
    }
}
